import pandas as pd
import matplotlib.pyplot as plt

plt.rcParams['font.family'] = 'Microsoft YaHei'
# 重新加载雪龙集团利润表数据
file_path_profit = '../雪龙集团利润表.xlsx'
df_profit = pd.read_excel(file_path_profit, header=None)

# 设置列名
df_profit.columns = ["项目"] + [2020, 2021, 2022, 2023]

# 转换数据为数值格式
for col in [2020, 2021, 2022, 2023]:
    df_profit[col] = pd.to_numeric(df_profit[col].replace(',', '', regex=True), errors='coerce')

# 提取营业收入和净利润的数据
revenue_data = df_profit[df_profit["项目"] == "营业总收入(元)"]
profit_data = df_profit[df_profit["项目"] == "净利润(元)"]

# 计算营业收入增长率（从2020年到2023年）
revenue_growth_rate = revenue_data.iloc[0, 1:].pct_change() * 100  # pct_change()计算增长率，*100转换为百分比

# 计算净利润增长率（从2020年到2023年）
profit_growth_rate = profit_data.iloc[0, 1:].pct_change() * 100  # pct_change()计算增长率，*100转换为百分比

# 绘制折线图
plt.figure(figsize=(10, 6))

# 绘制营业收入增长率折线图
plt.plot([2021, 2022, 2023], revenue_growth_rate[1:], marker='o', label="营业收入增长率", color='blue')

# 绘制净利润增长率折线图
plt.plot([2021, 2022, 2023], profit_growth_rate[1:], marker='o', label="净利润增长率", color='green')

# 添加标题和标签
plt.title("雪龙集团2020-2023年发展能力分析", fontsize=16)
plt.xlabel("年份", fontsize=12)
plt.ylabel("增长率（%）", fontsize=12)

# 添加图例和网格
plt.legend(loc="best", fontsize=10)
plt.grid(alpha=0.5)
plt.tight_layout()

plt.savefig('../分析图象/发展能力分析图表.png')
# 显示图表
plt.show()
